#!/usr/bin/env python3
"""
数据库更新脚本 - 更新字段类型以支持富文本内容
"""

import sys
import os
sys.path.append(os.path.dirname(os.path.abspath(__file__)))

from sqlalchemy import text
from app.db.database import engine

def update_database():
    """更新数据库表结构"""
    try:
        with engine.connect() as conn:
            # 更新 homepages 表的 content 字段
            print("更新 homepages 表的 content 字段...")
            conn.execute(text("""
                ALTER TABLE homepages 
                MODIFY COLUMN content LONGTEXT
            """))
            
            # 更新 product_details 表的 rich_content 字段
            print("更新 product_details 表的 rich_content 字段...")
            conn.execute(text("""
                ALTER TABLE product_details 
                MODIFY COLUMN rich_content LONGTEXT
            """))
            
            conn.commit()
            print("数据库更新完成！")
            
    except Exception as e:
        print(f"更新数据库时出错: {e}")
        return False
    
    return True

if __name__ == "__main__":
    print("开始更新数据库...")
    if update_database():
        print("✅ 数据库更新成功！")
    else:
        print("❌ 数据库更新失败！") 